【问题描述】
使用pandas将生成的数据写入到指定路径的Excel文件。
【示例2-3】
本例在示例2-2的基础上将导入的6行6列数据导出到相同路径下的“单品电商数据2.xlsx”中。
- 编写下面的代码：
code.python
import pandas as pd
# 读取Excel文件的第一个工作表，使用openpyxl引擎
df = pd.read_excel(r'D:/Samples/ch02/02 将数据写入Excel文件/单品电商数据.xlsx', engine='openpyxl')
# 获取Excel文件所在目录的完整路径
file_path = r'D:/Samples/ch02/02 将数据写入Excel文件/单品电商数据.xlsx'
directory = file_path[:file_path.rfind("/")]
# 取前6行，前6列数据
df_new = df.iloc[:6,:6]
# 将数据保存到该路径下的“单品电商数据2.xlsx”中
output_file = directory + r"/单品电商数据2.xlsx"
df_new.to_excel(output_file, index=False)打开Python IDLE，新建一个脚本文件，将上面生成的代码复制进去，保存。运行脚本，导入的数据被保存到相同路径下的“单品电商数据2.xlsx”文件中。打开该文件，效果如图2-2所示。
图2-2 将数据写入Excel文件
【知识点扩展】
使用DataFrame对象的to_excel方法将数据导出到指定的Excel文件。用excel_writer参数指定导出文件的路径和名称，用sheet_name参数指定工作表的名称。